package com.train.webadmin.app.service.impl;

import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.train.common.core.object.CallResult;
import com.train.webadmin.app.service.*;
import com.train.webadmin.app.dao.*;
import com.train.webadmin.app.model.*;
import com.train.common.core.base.dao.BaseDaoMapper;
import com.train.common.core.object.Tuple2;
import com.train.common.core.object.TokenData;
import com.train.common.core.object.MyRelationParam;
import com.train.common.core.base.service.BaseService;
import com.train.common.core.util.MyModelUtil;
import com.github.pagehelper.Page;
import lombok.extern.slf4j.Slf4j;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.*;

/**
 * 班级学生关系表数据操作服务类。
 *
 * @author xulei
 * @date 2025-03-07
 */
@Slf4j
@Service("classStudentService")
public class ClassStudentServiceImpl extends BaseService<ClassStudent, Long> implements ClassStudentService {

    @Autowired
    private ClassStudentMapper classStudentMapper;

    /**
     * 返回当前Service的主表Mapper对象。
     *
     * @return 主表Mapper对象。
     */
    @Override
    protected BaseDaoMapper<ClassStudent> mapper() {
        return classStudentMapper;
    }

    @Transactional(rollbackFor = Exception.class)
    @Override
    public ClassStudent saveNew(ClassStudent classStudent) {
        classStudentMapper.insert(this.buildDefaultValue(classStudent));
        return classStudent;
    }

    @Transactional(rollbackFor = Exception.class)
    @Override
    public void saveNewBatch(List<ClassStudent> classStudentList) {
        if (CollUtil.isNotEmpty(classStudentList)) {
            classStudentList.forEach(this::buildDefaultValue);
            classStudentMapper.insertList(classStudentList);
        }
    }

    @Transactional(rollbackFor = Exception.class)
    @Override
    public void saveNewOrUpdateBatch(List<ClassStudent> classStudentList, List<String> duplicateVerifyColumns, int batchSize) {
        if (CollUtil.isEmpty(classStudentList)) {
            return;
        }
        if (batchSize <= 0) {
            batchSize = CollUtil.isNotEmpty(duplicateVerifyColumns) ? 100 : 10000;
        }
        int start = 0;
        do {
            int end = Math.min(classStudentList.size(), start + batchSize);
            List<ClassStudent> subList = classStudentList.subList(start, end);
            if (CollUtil.isNotEmpty(duplicateVerifyColumns)) {
                Tuple2<List<ClassStudent>, List<ClassStudent>> t = this.deduceInsertOrUpdateList(subList, duplicateVerifyColumns);
                if (CollUtil.isNotEmpty(t.getFirst())) {
                    t.getFirst().forEach(this::buildDefaultValue);
                    classStudentMapper.insertList(t.getFirst());
                }
                t.getSecond().forEach(data -> classStudentMapper.updateById(data));
            } else {
                classStudentList.forEach(this::buildDefaultValue);
                classStudentMapper.insertList(subList);
            }
            if (end == classStudentList.size()) {
                break;
            }
            start += batchSize;
        } while (true);
    }

    @Transactional(rollbackFor = Exception.class)
    @Override
    public boolean update(ClassStudent classStudent, ClassStudent originalClassStudent) {
        classStudent.setUpdateUserId(TokenData.takeFromRequest().getUserId());
        classStudent.setCreateUserId(originalClassStudent.getCreateUserId());
        classStudent.setCreateTime(originalClassStudent.getCreateTime());
        classStudent.setUpdateTime(new Date());
        // 这里重点提示，在执行主表数据更新之前，如果有哪些字段不支持修改操作，请用原有数据对象字段替换当前数据字段。
        UpdateWrapper<ClassStudent> uw = this.createUpdateQueryForNullValue(classStudent, classStudent.getId());
        return classStudentMapper.update(classStudent, uw) == 1;
    }

    @Transactional(rollbackFor = Exception.class)
    @Override
    public boolean remove(Long id) {
        return classStudentMapper.deleteById(id) == 1;
    }

    @Override
    public List<ClassStudent> getClassStudentList(ClassStudent filter, String orderBy) {
        return classStudentMapper.getClassStudentList(filter, orderBy);
    }

    @Override
    public List<ClassStudent> getClassStudentListWithRelation(ClassStudent filter, String orderBy) {
        List<ClassStudent> resultList = classStudentMapper.getClassStudentList(filter, orderBy);
        // 在缺省生成的代码中，如果查询结果resultList不是Page对象，说明没有分页，那么就很可能是数据导出接口调用了当前方法。
        // 为了避免一次性的大量数据关联，规避因此而造成的系统运行性能冲击，这里手动进行了分批次读取，开发者可按需修改该值。
        int batchSize = resultList instanceof Page ? 0 : 1000;
        this.buildRelationForDataList(resultList, MyRelationParam.normal(), batchSize);
        return resultList;
    }

    private ClassStudent buildDefaultValue(ClassStudent classStudent) {
        TokenData tokenData = TokenData.takeFromRequest();
        classStudent.setUpdateUserId(tokenData.getUserId());
        classStudent.setCreateUserId(tokenData.getUserId());
        Date now = new Date();
        classStudent.setCreateTime(now);
        classStudent.setUpdateTime(now);
        MyModelUtil.setDefaultValue(classStudent, "status", 0);
        return classStudent;
    }

    @Override
    public List<ClassStudent> getClassStudentListByClassId(Long classId) {
        return classStudentMapper.getClassStudentListByClassId(classId);
    }

    @Override
    public ClassStudent getClassStudent(Long schoolId, String studentName, Long classId) {
        return classStudentMapper.getClassStudent(schoolId, studentName, classId);
    }

    @Override
    public boolean updateClassStudent(ClassStudent classStudent) {
        return classStudentMapper.updateClassStudent(classStudent);
    }

    @Override
    public Integer countByClassId(Long classId) {
        return classStudentMapper.countByClassId(classId);
    }
}
